import unittest
from selenium import webdriver


class JXC_Login_Test(unittest.TestCase):

    def test_login_error(self):
        driver = webdriver.Firefox()
        driver.get('http://localhost/myweb/jxc/index.asp')
        # 输入错误的用户名和密码
        driver.find_element_by_name('username').send_keys('abcd')
        driver.find_element_by_name('pwd').send_keys('12234')
        driver.find_element_by_name('enter').click()

        # 获取提示框中的提示文字
        alert = driver.switch_to.alert.text
        # 断言：应该弹出错误的提示信息
        # 提示信息中应该出现“。。。。密码错误!”这几个字
        self.assertIn('密码错误', alert)

    def test_login_ok(self):
        driver = webdriver.Firefox()
        driver.get('http://localhost/myweb/jxc/index.asp')
        driver.find_element_by_name('username').send_keys('admin')
        driver.find_element_by_name('pwd').send_keys('admin')
        driver.find_element_by_name('enter').click()

        # 断言：登录是否成功
        #  依据：'后台管理系统'这几个字，出现在页面源代码中
        self.assertIn('后台管理系统', driver.page_source)


unittest.main()

